613b414c1c6ad328d8ba0d2a1107ef617582143a,python/src/com/jetbrains/python/psi/impl/PyTargetExpressionImpl.java,PyTargetExpressionImpl,getType,#TypeEvalContext#,100
Before Change
if (cls != null) {
final PyFunction enter = cls.findMethodByName(PyNames.ENTER, true);
if (enter != null) {
return enter.getReturnType(context, null);
}
}
}
After Change
if (cls != null) {
final PyFunction enter = cls.findMethodByName(PyNames.ENTER, true);
if (enter != null) {
final PyType enterType = enter.getReturnType(context, null);
if (enterType != null) {
return enterType;
}
// Guess the return type of __enter__
return PyUnionType.createWeakType(exprType);
}
}
}